# __init__.py 为初始化加载文件
import time
from typing import Optional, List

from ascript.android import plug, system, action
from ascript.android.node import Selector
from ld.common.CommonClass import CommonResult

plug.load("ld:1.2.0") # 这里是版本号


from .toutiao_option import 流程获取下拉词操作, 打印设备基本信息
from .特征库 import *
from .关键词和关键词结果 import KeywordProcessor

# 打开头条
system.open("今日头条")

打印设备基本信息(ld)

# 1.获取要搜索的关键词。2.获取一个下拉词的结果。3.保存下拉词结果到文件。执行第2步获取下一个关键词直到结束。

# 获取当前项目的目录
import os
current_dir = os.path.dirname(os.path.abspath(__file__))

关键词管理器 = KeywordProcessor(
    keyword_file=os.path.join(current_dir, "keywords.txt"),
    result_file=os.path.join(current_dir, "./results.txt"),
    export_file=os.path.join(current_dir, "./output.txt")
)

# 获取断点继续的关键词
关键词s: Optional[List[str]] = 关键词管理器.get_todo_keywords()
for 关键词 in 关键词s:
    # 获取下拉词
    下拉词: Optional[List[str]] = 流程获取下拉词操作(关键词, ld)
    if 下拉词:
        log.info(f"保存{关键词}的下拉词")
        关键词管理器.save_results(关键词, 下拉词)
    else:
        print(f"未获取到{关键词}下拉词")
    time.sleep(2)

关键词管理器.export()
log.info("🎶🎶下拉词导出完成👌🎉🎉🎉")
